package com.wuzuqing.android.eightdiagrams.term;

import java.util.*;

public class LunarYear {
    public static int[] _LEAP_11 = new int[]{75, 94, 170, 238, 265, 322, 389, 469, 553, 583, 610, 678, 735, 754, 773,
            849, 887, 936, 1050, 1069, 1126, 1145, 1164, 1183, 1259, 1278, 1308, 1373, 1403, 1441, 1460, 1498, 1555,
            1593, 1612, 1631, 1642, 2033, 2128, 2147, 2242, 2614, 2728, 2910, 3062, 3244, 3339, 3616, 3711, 3730,
            3825, 4007, 4159, 4197, 4322, 4341, 4379, 4417, 4531, 4599, 4694, 4713, 4789, 4808, 4971, 5085, 5104,
            5161, 5180, 5199, 5294, 5305, 5476, 5677, 5696, 5772, 5791, 5848, 5886, 6049, 6068, 6144, 6163, 6258,
            6402, 6440, 6497, 6516, 6630, 6641, 6660, 6679, 6736, 6774, 6850, 6869, 6899, 6918, 6994, 7013, 7032,
            7051, 7070, 7089, 7108, 7127, 7146, 7222, 7271, 7290, 7309, 7366, 7385, 7404, 7442, 7461, 7480, 7491,
            7499, 7594, 7624, 7643, 7662, 7681, 7719, 7738, 7814, 7863, 7882, 7901, 7939, 7958, 7977, 7996, 8034,
            8053, 8072, 8091, 8121, 8159, 8186, 8216, 8235, 8254, 8273, 8311, 8330, 8341, 8349, 8368, 8444, 8463,
            8474, 8493, 8531, 8569, 8588, 8626, 8664, 8683, 8694, 8702, 8713, 8721, 8751, 8789, 8808, 8816, 8827,
            8846, 8884, 8903, 8922, 8941, 8971, 9036, 9066, 9085, 9104, 9123, 9142, 9161, 9180, 9199, 9218, 9256,
            9294, 9313, 9324, 9343, 9362, 9381, 9419, 9438, 9476, 9514, 9533, 9544, 9552, 9563, 9571, 9582, 9601,
            9639, 9658, 9666, 9677, 9696, 9734, 9753, 9772, 9791, 9802, 9821, 9886, 9897, 9916, 9935, 9954, 9973, 9992};
    public static int[] _LEAP_12 = new int[]{37, 56, 113, 132, 151, 189, 208, 227, 246, 284, 303, 341, 360, 379, 417,
            436, 458, 477, 496, 515, 534, 572, 591, 629, 648, 667, 697, 716, 792, 811, 830, 868, 906, 925, 944, 963,
            982, 1001, 1020, 1039, 1058, 1088, 1153, 1202, 1221, 1240, 1297, 1335, 1392, 1411, 1422, 1430, 1517, 1525
            , 1536, 1574, 3358, 3472, 3806, 3988, 4751, 4941, 5066, 5123, 5275, 5343, 5438, 5457, 5495, 5533, 5552,
            5715, 5810, 5829, 5905, 5924, 6421, 6535, 6793, 6812, 6888, 6907, 7002, 7184, 7260, 7279, 7374, 7556,
            7746, 7757, 7776, 7833, 7852, 7871, 7966, 8015, 8110, 8129, 8148, 8224, 8243, 8338, 8406, 8425, 8482,
            8501, 8520, 8558, 8596, 8607, 8615, 8645, 8740, 8778, 8835, 8865, 8930, 8960, 8979, 8998, 9017, 9055,
            9074, 9093, 9112, 9150, 9188, 9237, 9275, 9332, 9351, 9370, 9408, 9427, 9446, 9457, 9465, 9495, 9560,
            9590, 9628, 9647, 9685, 9715, 9742, 9780, 9810, 9818, 9829, 9848, 9867, 9905, 9924, 9943, 9962, 10000};

    public static Map<String, LunarYear> _CACHE = new HashMap<>();
    public static Map<String, Integer> _LEAP = new HashMap<>();
    public int year;
    public List<LunarMonth> months;
    public List<Double> jieQiJulianDays;

    static {
        _initLeap();
    }

    public static void _initLeap() {
        int i, j;
        for (i = 0, j = _LEAP_11.length; i < j; i++) {
            _LEAP.put("_" + _LEAP_11[i], 13);
        }
        for (i = 0, j = _LEAP_12.length; i < j; i++) {
            _LEAP.put("_" + _LEAP_12[i], 14);
        }
    }

    public LunarYear(int year) {
        this.year = year;
        _compute();
    }

    public static LunarYear _fromYear(int lunarYear) {
        return new LunarYear(lunarYear);
    }

    public static LunarYear _fromCachedYear(int lunarYear) {
        String key = "_" + lunarYear;
        return _CACHE.computeIfAbsent(key, k -> _fromYear(lunarYear));
    }


    public static LunarYear fromYear(int lunarYear) {
        return _fromCachedYear(lunarYear);
    }


    public int getYear() {
        return this.year;
    }

    public List<Double> getJieQiJulianDays() {
        return this.jieQiJulianDays;
    }

    public List<LunarMonth> getMonths() {
        return this.months;
    }

    public LunarMonth getMonth(int lunarMonth) {
        for (int i = 0, j = this.months.size(); i < j; i++) {
            LunarMonth m = this.months.get(i);
            if (m.getYear() == this.year && m.getMonth() == lunarMonth) {
                return m;
            }
        }
        return null;
    }

    public int getLeapMonth() {
        for (int i = 0, j = this.months.size(); i < j; i++) {
            LunarMonth m = this.months.get(i);
            if (m.getYear() == this.year && m.isLeap()) {
                return Math.abs(m.getMonth());
            }
        }
        return 0;
    }

    @Override
    public String toString() {
        return this.getYear() + "";
    }

    public String toFullString() {
        return this.getYear() + "年";
    }

    public LunarYear _compute() {
        this.months = new ArrayList<>();
        this.jieQiJulianDays = new ArrayList<>();
        // 节气(中午12点)，长度25
        int[] jq = new int[25];
        // 合朔，即每月初一(中午12点)，长度16
        List<Integer> hs = new ArrayList<>();
        // 每月天数，长度15
        List<Integer> dayCounts = new ArrayList<>();
        int i, j;
        int currentYear = this.year;
        int year = currentYear - 2000;
        // 从上年的大雪到下年的大寒
        for (i = 0, j = Lunar.JIE_QI_IN_USE.length; i < j; i++) {
            // 精确的节气
            double t = 36525 * ShouXingUtil.saLonT((year + (17 + i) * 15.0 / 360) * ShouXingUtil.PI_2);
            t += ShouXingUtil.ONE_THIRD - ShouXingUtil.dtT(t);
            this.jieQiJulianDays.add((t + Solar.J2000));
            // 按中午12点算的节气
            if (i > 0 && i < 26) {
                jq[i - 1] = (int) Math.round(t);
            }
        }
//        System.out.println(Arrays.toString(jq));
        // 冬至前的初一
        double w = ShouXingUtil.calcShuo(jq[0]);
        if (w > jq[0]) {
            w -= 29.5306;
        }
//        System.out.println(BigDecimal.valueOf(w));
        // 递推每月初一
        for (i = 0; i < 16; i++) {
            hs.add((int) ShouXingUtil.calcShuo((w + 29.5306 * i)));
        }
//        System.out.println(hs);
        // 每月天数
        for (i = 0; i < 15; i++) {
            dayCounts.add((int) Math.floor(hs.get(i + 1) - hs.get(i)));
        }
//        System.out.println(dayCounts);
        Integer currentYearLeap = _LEAP.get("_" + currentYear);
        if (currentYearLeap == null) {
            currentYearLeap = -1;
            if (hs.get(13) <= jq[24]) {
                i = 1;
                while (hs.get(i + 1) > jq[2 * i] && i < 13) {
                    i++;
                }
                currentYearLeap = i;
            }
        }

        int prevYear = currentYear - 1;
        Integer prevYearLeap = _LEAP.get("_" + prevYear);
        prevYearLeap = prevYearLeap != null ? prevYearLeap - 12 : -1;

        int y = this.year - 1;
        int m = 11;
        for (i = 0,
                     j = dayCounts.size(); i < j; i++) {
            int cm = m;
            boolean isNextLeap = false;
            if (y == currentYear && i == currentYearLeap) {
                cm = -cm;
            } else if (y == prevYear && i == prevYearLeap) {
                cm = -cm;
            }
            if (y == currentYear && i + 1 == currentYearLeap) {
                isNextLeap = true;
            } else if (y == prevYear && i + 1 == prevYearLeap) {
                isNextLeap = true;
            }
            this.months.add(LunarMonth._(y, cm, dayCounts.get(i), hs.get(i) + Solar.J2000));
            if (!isNextLeap) {
                m++;
            }
            if (m == 13) {
                m = 1;
                y++;
            }
        }
        return this;
    }

}
